#pragma once
#include <iostream>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <fcntl.h>
#include <string>
#include "common.hpp"
// #define FIFO_FILE "fifo"

#define PATH "."
#define NAME "fifo"

class NamedPipe
{
public:
    NamedPipe(const std::string path, const std::string name)
        : _path(path), _name(name)
    {
        // 创建命名管道文件
        _pn = _path + "/" + _name;
        umask(0);
        int n = mkfifo(_pn.c_str(), 0666);
        if (n == 0)
        {
            std::cout << "mkfifo success." << std::endl;
        }
        else
        {
            // std::cerr << "mkfifo fail." << std::endl;
            ERR_EXIT("mkfifo");
        }
    }
    ~NamedPipe()
    {
        // 删除命名管道
        int n = unlink(_pn.c_str());
        if (n == 0)
        {
            std::cout << "unlink fifo success" << std::endl;
        }
        else
        {
            // std::cerr << "unlink fifo fail" << std::endl;
            ERR_EXIT("unlink");
        }
    }

private:
    std::string _path;
    std::string _name;
    std::string _pn;
};

class PipeOperator
{
public:
    PipeOperator(const std::string path, const std::string name)
        : _path(path), _name(name), _fd(-1)
    {
        _pn = _path + "/" + _name;
    }
    void OpenForRead()
    {
        // 打开管道文件
        _fd = open(_pn.c_str(), O_RDONLY);
        if (_fd == -1)
        {
            // std::cerr << "open fifo fail." << std::endl;
            ERR_EXIT("open");
        }
        else
        {
            std::cout << "open fifo success." << std::endl;
        }
    }
    void OpenForWrite()
    {
        // 打开管道文件准备写
        _fd = open(_pn.c_str(), O_WRONLY);
        if (_fd == -1)
        {
            // std::cerr << "open fifo fail." << std::endl;
            ERR_EXIT("open");
        }
        else
        {
            std::cout << "open fifo success." << std::endl;
        }
    }
    bool Wait()
    {
        char c;
        int n = read(_fd, &c, 1);
        if (n > 0)
        {
            std::cout << "Wait success" << std::endl;
            return true;
        }
        return false;
        // // 进行通信：从管道文件中读取
        // while (true)
        // {
        //     char buffer[1024];
        //     int n = read(_fd, buffer, sizeof(buffer) - 1);
        //     if (n > 0)
        //     {
        //         buffer[n] = 0; // 保证字符串安全
        //         std::cout << "read fifo success." << std::endl;
        //         std::cout << "cilent say# " << buffer << std::endl;
        //     }
        //     else if (n == 0)
        //     {
        //         std::cout << "cilent quit! me too." << std::endl;
        //         break;
        //     }
        //     else
        //     {
        //         // std::cerr << "read fifo fail" << std::endl;
        //         ERR_EXIT("read");
        //         break;
        //     }
        // }
    }
    void Wakeup()
    {
        // 写入内容
        char c = 'c';
        int n = write(_fd, &c, 1);
        if (n == -1)
        {
            ERR_EXIT("write");
        }
        else
        {
            std::cout << "Wakeup success" << std::endl;
        }
        // while (true)
        // {
        //     std::string message;
        //     std::cout << "Please enter# ";
        //     std::getline(std::cin, message);
        //     pid_t id = getpid();
        //     message += ", message number: " + std::to_string(cnt++) + ", pid: [" + std::to_string(id) + "]";
        //     int n = write(_fd, message.c_str(), message.size());
        //     if (n == -1)
        //     {
        //         // std::cerr << "write fail" << std::endl;
        //         ERR_EXIT("write");
        //     }
        //     else
        //     {
        //         std::cout << "write success" << std::endl;
        //     }
        // }
    }
    void Close()
    {
        if (_fd > 0)
            close(_fd);
    }
    ~PipeOperator() {}

private:
    std::string _path;
    std::string _name;
    std::string _pn;
    int _fd;
};